package recursion;

/**
 * 8 皇后问题 (这里用的是回溯算法，效率并不是很高，后面会使用贪心算法进行优化) 92 种
 * @Auther Jun jie Yi
 * @Date 15:58 2021/7/11
 */
public class Queue8 {
    //定义一个 max 表示共有多少个皇后
    int max = 8;
    //定义数组 array，保存皇后放置位置的结果，比如 arr = {0, 4, 7, 5, 2, 6, 1, 3}
    int[] array = new int[max];
    static int count = 0;
    static int judgeCount = 0;
    public static void main(String[] args) {
        Queue8 queue8 = new Queue8();
        queue8.check(0);
        System.out.println(count);
        System.out.println(judgeCount);
    }

    //编写一个方法，放置第n个皇后
    private void check(int n) {
        //n==8，8皇后已经放好了
        if(n == max) {
            print();
            return;
        }
        //依次放入皇后，并判断是否冲突
        for (int i = 0; i < max; i++) {
            //先把当前这个皇后
            array[n] = i;
            //当放置第n个皇后到i列时，是否冲突
            if(judge(n)) {//如果不冲突(如果冲突就继续往后面的列走)
                //接着放n + 1个皇后，即开始递归
                check(n + 1);
            }
        }
    }

    /**
     * 当我们放置第n个皇后后，就去检查该皇后是否和前面已经摆放的皇后冲突
     * n - i == Math.abs(array[n] - array[i]) 判断是否在一个斜线上，这里有难度(这里是一个等腰三角形)
     * @param n 表示第n个皇后
     * @return
     */
    private boolean judge(int n) {
        judgeCount ++;
        for(int i = 0; i < n; i++) {
            if(array[i] == array[n] || n - i == Math.abs(array[n] - array[i])) {
                return false;
            }
        }
        return true;
    }

    //写一个方法，将皇后摆放的位置输出
    private void print() {
        for (int i = 0; i < array.length; i++) {
            System.out.print(array[i] + " ");
        }
        count ++;
        System.out.println();
    }
}
